package org.bright;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.baomidou.mybatisplus.generator.fill.Column;
import com.baomidou.mybatisplus.generator.fill.Property;
import org.bright.entity.BaseEntity;

import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

public class CodeGenerator {
    public static void main(String[] args) {
        // 使用 FastAutoGenerator 快速配置代码生成器
        // 数据库配置
        FastAutoGenerator.create("jdbc:mysql://localhost:3306/749shopping",
                        "root", "868819qwe")
                // 全局配置
                .globalConfig(builder -> {
                    builder.author("bright") // 设置作者
                            .dateType(DateType.ONLY_DATE)// 设置时间类型策略
                            .commentDate("yyyy-MM-dd hh:MM") // 设置注释日期格式
                            .enableSpringdoc()
                            .outputDir("src/main/java"); // 输出目录
                })
//                配置文件生成能覆盖之前的文件


                // 包配置
                .packageConfig(builder -> {
                    builder.parent("org.bright") // 设置父包名
                            .entity("entity") // 设置实体类包名
                            .mapper("mapper") // 设置 Mapper 接口包名
                            .service("service") // 设置 Service 接口包名
                            .serviceImpl("service.impl") // 设置 Service 实现类包名
                            .controller("controller") // 设置 Controller 包名
                            .xml("mapper"); // 设置 Mapper XML 文件包名
                })
                // 策略配置
                .strategyConfig(builder -> {
                    builder.addInclude(getAllTableNames()) // 设置需要生成的表名
                            .entityBuilder()
                            .superClass(BaseEntity.class)// 父类
                            .disableSerialVersionUID()// 禁用生成 serialVersionUID
                            .naming(NamingStrategy.underline_to_camel)// 数据库表映射到实体的命名策略
                            .columnNaming(NamingStrategy.underline_to_camel)// 下划线转驼峰命名
                            .addSuperEntityColumns("id", "created_by", "created_at", "updated_by", "updated_at")// 添加父类公共字段
                            .enableLombok() // 启用 Lombok
                            .enableTableFieldAnnotation() // 启用字段注解
                            .enableFileOverride()
                            .idType(IdType.AUTO)
                            .controllerBuilder()
                            .enableRestStyle() // 启用 REST 风格
                            .enableFileOverride()
                            .mapperBuilder().enableFileOverride()
                            .serviceBuilder().enableFileOverride();


                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用 Freemarker 模板引擎
                .execute(); // 执行生成
    }

    /**
     * 获取数据库中的所有表名
     *
     * @return 表名列表
     */
    private static String[] getAllTableNames() {
        List<String> tableNames = new ArrayList<>();
        try (Connection connection = DriverManager.getConnection("jdbc:mysql://localhost:3306/749shopping", "root", "868819qwe")) {
            DatabaseMetaData metaData = connection.getMetaData();
            ResultSet resultSet = metaData.getTables(null, null, "%", new String[]{"TABLE"});
            while (resultSet.next()) {
                tableNames.add(resultSet.getString("TABLE_NAME"));
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return tableNames.toArray(new String[0]);
    }

    /**
     * <p>
     * 读取控制台内容
     * </p>
     */
//    public static String scanner(String tip) {
//        Scanner scanner = new Scanner(System.in);
//        StringBuilder help = new StringBuilder();
//        help.append("请输入" + tip + "：");
//        System.out.println(help.toString());
//        if (scanner.hasNext()) {
//            String ipt = scanner.next();
//            if (StringUtils.isNotEmpty(ipt)) {
//                return ipt;
//            }
//        }
//        throw new MybatisPlusException("请输入正确的" + tip + "！");
//    }
}
